![Athena-ExpressでAmazon Athena Parameterized Queriesを実行する](https://devio2023-media.developers.io/wp-content/uploads/2022/08/amazon-athena.png)
Athena-ExpressでAmazon Athena Parameterized Queriesを実行する
こんにちは、CX事業本部 IoT事業部の若槻です。
前回のエントリでAthena-Expressを使ってAmazon Athenaへのクエリの実行と結果取得を行いましたが、その際に「Athena-ExpressではParameterized Queriesを実行できるのか?」が気になったので確認してみました。(結論としては実行できました)
Athena-ExpressではExecutionParametersを指定できない
Parameterized QueriesをAPIから実行したい場合は、次のようにStartQueryExecutionでExecutionParametersを指定します。
aws athena start-query-execution \ --query-string "SELECT * FROM gluedatabase.source_glue_table WHERE deviceId = ?" \ --execution-parameters "1" "2"
しかしAthena-ExpressのパラメータではStartQueryExecution APIのExecutionParametersに該当するパラメータがありません。
//StartQueryExecutionのExecutionParametersに該当するパラメータがない let myQuery = { sql: "SELECT * FROM elb_logs LIMIT 3" // required, db: "sampledb", // optional. pagination: 5, //optional NextToken: "ARfCDXRjMk...", //optional QueryExecutionId: "c274843b-4c5c-4ccf-ac8b-e33d595b927d", //optional catalog: "hive" //optional };
この場合に、Athena-ExpressでParameterized Queriesを実行する方法はあるのでしょうか。
実行する方法
Parameterized Queriesを予めPrepared Statementとして作成しておき、EXECUTE ... USING
というステートメントを使用すればAthena-ExpressでもParameterized Queriesを実行することができました。
Prepared Statementの事前作成
実行したいParameterized QueriesのPrepared Statementを事前に作成しておきます。
STATEMENT_NAME=queryDevices WORK_GROUP=workGroupVersionSpecified QUERY_STATEMENT="SELECT * FROM gluedatabase.source_glue_table WHERE deviceId = ?" aws athena create-prepared-statement \ --statement-name ${STATEMENT_NAME} \ --work-group ${WORK_GROUP} \ --query-statement ${QUERY_STATEMENT}
Athena-ExpressでParameterized Queriesを実行する
EXECUTE ... USING
のステートメントをAthena-Expressで実行します。
import { AthenaExpress } from 'athena-express'; import * as AWS from 'aws-sdk'; AWS.config.update({ region: 'ap-northeast-1' }); const athenaExpressConfig = { aws: AWS, //必須 workgroup: 'workGroupVersionSpecified', //オプション。既定はdefault }; const athenaExpress = new AthenaExpress(athenaExpressConfig); const main = async () => { const sqlQuery = "EXECUTE queryDevices USING 'd001'"; const results = await athenaExpress.query(sqlQuery); console.log(results); }; main();
するとParameterized Queriesのクエリを実行でき、結果を取得することができました。
$ npx ts-node athena-express-query.ts { Items: [ { deviceid: 'd001', maxtemperature: 19.8, year: 2022, month: 10, day: 30 }, { deviceid: 'd001', maxtemperature: 22.7, year: 2022, month: 10, day: 15 }, { deviceid: 'd001', maxtemperature: 19.8, year: 2022, month: 11, day: 1 }, { deviceid: 'd001', maxtemperature: 19.1, year: 2022, month: 10, day: 1 } ] }
おわりに
Athena-ExpressでAmazon Athena Parameterized Queriesを実行してみました。
前回のエントリでお伝えしたとおり、Athena-Expressを使うとAmazon Athenaへのクエリの実行と結果取得をシンプルにすることができます。Parameterized Queriesを使用すると予めクエリの内容のパターンが決まっているクエリの呼び出しをさらにシンプルにすることができ、またSQLインジェクション対策にもなります。Prepared Statementを事前作成しておく必要はありますが、Parameterized QueriesをAthena-Expressでも使えるのは両者のいいところ取りができて助かりますね。
参考
- create-prepared-statement — AWS CLI 2.8.5 Command Reference
- AWS CDKでAmazon Athena Parameterized Queriesを作ってみた | DevelopersIO
- /sql
以上